package files

import (
	"github.com/flipped-aurora/gin-vue-admin/server/global"
	"github.com/flipped-aurora/gin-vue-admin/server/model/common/request"
	"github.com/flipped-aurora/gin-vue-admin/server/model/files"
	filesReq "github.com/flipped-aurora/gin-vue-admin/server/model/files/request"
)

type FileTypeService struct {
}

// CreateFileType 创建文件类型记录
// Author [piexlmax](https://github.com/piexlmax)
func (fileTypeService *FileTypeService) CreateFileType(fileType *files.FileType) (err error) {
	err = global.GVA_DB.Create(fileType).Error
	return err
}

// DeleteFileType 删除文件类型记录
// Author [piexlmax](https://github.com/piexlmax)
func (fileTypeService *FileTypeService) DeleteFileType(fileType files.FileType) (err error) {
	err = global.GVA_DB.Delete(&fileType).Error
	return err
}

// DeleteFileTypeByIds 批量删除文件类型记录
// Author [piexlmax](https://github.com/piexlmax)
func (fileTypeService *FileTypeService) DeleteFileTypeByIds(ids request.IdsReq) (err error) {
	err = global.GVA_DB.Delete(&[]files.FileType{}, "id in ?", ids.Ids).Error
	return err
}

// UpdateFileType 更新文件类型记录
// Author [piexlmax](https://github.com/piexlmax)
func (fileTypeService *FileTypeService) UpdateFileType(fileType files.FileType) (err error) {
	err = global.GVA_DB.Save(&fileType).Error
	return err
}

// GetFileType 根据id获取文件类型记录
// Author [piexlmax](https://github.com/piexlmax)
func (fileTypeService *FileTypeService) GetFileType(id uint) (fileType files.FileType, err error) {
	err = global.GVA_DB.Where("id = ?", id).First(&fileType).Error
	return
}

// GetFileTypeInfoList 分页获取文件类型记录
// Author [piexlmax](https://github.com/piexlmax)
func (fileTypeService *FileTypeService) GetFileTypeInfoList(info filesReq.FileTypeSearch) (list []files.FileType, total int64, err error) {
	limit := info.PageSize
	offset := info.PageSize * (info.Page - 1)
	// 创建db
	db := global.GVA_DB.Model(&files.FileType{})
	var fileTypes []files.FileType
	// 如果有条件搜索 下方会自动创建搜索语句
	if info.StartCreatedAt != nil && info.EndCreatedAt != nil {
		db = db.Where("created_at BETWEEN ? AND ?", info.StartCreatedAt, info.EndCreatedAt)
	}
	if info.TypeName != "" {
		db = db.Where("type_name = ?", info.TypeName)
	}
	err = db.Count(&total).Error
	if err != nil {
		return
	}

	if limit != 0 {
		db = db.Limit(limit).Offset(offset)
	}

	err = db.Find(&fileTypes).Error
	return fileTypes, total, err
}
